package model

import (
	"database/sql"
	"fmt"
	"time"
	"wlblog/src/utils"

	"gorm.io/driver/mysql"
	"gorm.io/driver/postgres"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

var (
	db *gorm.DB
	Db  *sql.DB
	err error
)

func InitDb() {
	switch utils.DbType {
	case "mysql":
		dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
			utils.MysqlUser,
			utils.MysqlPassword,
			utils.MysqlHost,
			utils.MysqlPort,
			utils.MysqlDbName,
		)
		db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	case "postgres":
		dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=%s",
			utils.PgHost,
			utils.PgUser,
			utils.PgPassword,
			utils.PgDbName,
			utils.PgPort,
			utils.PgSslMode,
			utils.PgTimeZone,
		)
		db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
	case "sqlite":
		db, err = gorm.Open(sqlite.Open(utils.SqliteFile), &gorm.Config{})
	default:
		panic(fmt.Sprintf("不支持的数据库类型: %s", utils.DbType))
	}

	if err != nil {
		panic(fmt.Sprintf("连接数据库失败: %v", err))
	}

	if Db, err = db.DB(); err != nil {
		panic(fmt.Sprintf("创建连接数池失败: %v", err))
	}

	db.AutoMigrate(&User{}, &Category{}, &Article{})

	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	Db.SetMaxIdleConns(10)

	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	Db.SetMaxOpenConns(100)

	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	Db.SetConnMaxLifetime(time.Second * 10)
}
